from django.shortcuts import render

#导包
from django.http import HttpResponse,HttpResponseRedirect
#导入类路由库
from django.views import View
#导入数据库类
from mysite.models import User
from supermarket.models import Product

import json

import time
#导包
from django.http import HttpResponse,HttpResponseRedirect
#模板模块
from django.shortcuts import render
#导入数据模型
from mysite.models import User
#导入绘图库
from PIL import ImageDraw
#导入绘图字体库
from PIL import ImageFont
#导入图片库
from PIL import Image
#导入io库
import io
#导入随机库
import random

#导入钻饰器
from mysite.d7 import check_login


#定义随机颜色方法
def get_random_color():
    R = random.randrange(255)
    G = random.randrange(255)
    B = random.randrange(255)

    return (R,G,B)

#定义随机验证码
def test_captcha(request):
    #定义背景颜色
    bg_color =get_random_color()
    #定义画布大小 宽 高
    img_size = (150,80)
    #定义画笔 颜色种类 画布 背景颜色
    image = Image.new("RGB",img_size,bg_color)
    #定义画笔对象 图片对象 颜色类型
    draw  = ImageDraw.Draw(image,'RGB')
    #定义随机字符
    source = '0123456789asdfghjkl'
    #定义4个字符
    #定义好容器，用来接收随机字符串
    code_str = ''
    for i in range(4):
        #获取随机颜色 字体颜色
        text_color = get_random_color()
        #获取随机字符串
        tmp_num = random.randrange(len(source))
        #获取字符集
        random_str = source[tmp_num]
        #将随机生成的字符串添加到容器中
        code_str += random_str
        #将字符画到画布上 坐标 字符串 字符串颜色 字体
        # #导入系统真实字体
        # my_font = ImageFont.truetype("路径")
        draw.text((10+30*i,20),random_str,text_color,font=None)
    #使用io获取一个缓冲区
    buf = io.BytesIO()
    #将图片保存到缓冲区
    image.save(buf,'png')

    #将随机码存到session中
    request.session['code'] = code_str

    #第二个参数声明头部信息
    return HttpResponse(buf.getvalue(),'image/png')


#定义商品列表页
class Prolist(View):
    #定义商品列表方法
    def get(self,request):
        #读取数据库
        res = Product.objects.all()
        #获取用户名
        username = request.COOKIES.get('username','未登录')

        #对用户名进行解码
        username = username.encode("ISO-8859-1").decode("UTF-8")

        #解析模板
        return render(request,'supermarket/prolist.html',locals())




#定义登陆逻辑
class Login(View):
    #定义登陆方法
    def post(self,request):
        time.sleep(2)
        #接收参数
        username = request.POST.get('username','未接收到username')
        password = request.POST.get('password','未接收到password')
        code = request.POST.get('code','未接收到code值')

        #判断用户名和密码
        res = User.objects.filter(username=username,password=password).count()
        if res > 0 and code == request.session['code']:
            #登陆成功逻辑
            response = HttpResponse('登陆成功')
            #对中文进行编码
            username = bytes(username,'utf-8').decode('ISO-8859-1')
            response.set_cookie("username",username,max_age=3600)
            response.set_cookie("password",password,max_age=3600)
            return response
        elif res > 0 and code != request.session['code']:
            return HttpResponse('验证码输入有误，请重新输入！')

        else:
            #登陆失败的逻辑
            return HttpResponse('您的用户名或者密码错误')

#定义注销功能
class Logout(View):
    #定义注销方法
    def get(self,request):
        #建立response对象
        response = HttpResponseRedirect("/supermarket/login")
        #删除cookie
        response.delete_cookie('username')
        return response

#定义删除操作
class ProDel(View):
    #定义删除方法
    def post(self,request):
        #接收参数
        id = request.POST.get('id')
        #删除操作
        Product.objects.filter(id=int(id)).delete()
        return HttpResponse("删除成功")

#定义商品编辑操作
class ProEdit(View):
    #定义编辑方法
    def post(self,request):
        #接收两个参数
        id = request.POST.get("id")
        price = request.POST.get("price")

        #进行修改操作
        Product.objects.filter(id=int(id)).update(price=int(price))

        return HttpResponse("修改成功")

#定义添加购物车逻辑
class AddCart(View):
    #定义添加方法
    def post(self,request):
        #接收参数
        id =request.POST.get('id')
        #判断购物车是否存在
        cart_list = request.session.get('cartlist','no')
        #如果不存在就创建后添加
        if cart_list == "no":
            clist = []
            clist.append(id)
            request.session['cartlist'] = clist
        else:
        #如果存在就直接添加
            request.session['cartlist'].append(id)

        return HttpResponse('添加成功')

#定义购物车列表
class CartList(View):
    #定义列表方法
    def get(self,request):
        #读取session
        cartlist = request.session.get('cartlist',0)

        return render(request,'supermarket/cartlist.html',locals())









#定义注册逻辑
class Reg(View):

    #定义注册接收方法
    def post(self,request):

        time.sleep(5)
        #接收参数
        username = request.POST.get('username','未接收到username')
        password = request.POST.get('password','未接收到password')
        
        #入库逻辑
        #验证唯一性
        res = User.objects.filter(username=username).count()
        if res > 0:
            #强转为json数据
            data = json.dumps({'msg':'请您换一个用户名'},ensure_ascii=False)
            return HttpResponse(data,content_type='application/json')
        else:
            #进行入库
            #实例化一个对象
            user = User(username=username,password=password)
            #保存
            user.save()
            data = json.dumps({'msg':'恭喜您，注册成功'},ensure_ascii=False)
            return HttpResponse(data,content_type='application/json')
# Create your views here.
